# coding=utf-8
# 代码文件：chapter5/ch5.5.py

import pandas as pd

f = r'data/北京房价数据.xlsx'

df = pd.read_excel(f)

# pivot_table函数获得DataFrame对象
df1 = df.pivot_table(index=['城区'], aggfunc='max', values='单价（元）')
# 过滤掉北京周边的数据
df2 = df1.query("城区!= '北京周边'")

# 对DataFrame对象进行排序
df3 = df2.sort_values(by='单价（元）', ascending=False)

f = r'data/temp.csv'

df3.to_csv(f, encoding='gbk')

print('Game Over')
